"""
# 实际结果如下：
# 登陆成功返回的实际结果如下：
login_ok_result = {
    "success": True,
    "code": 10000,
    "message": "操作成功！"
}

# 登陆失败返回的实际结果如下：
login_err_result = {
    "success": False,
    "code": 20001,
    "message": "用户名或密码错误"
}

1. 利用unittest框架，设计测试用例
2. 完成断言
"""
import unittest

login_ok_result = {
    "success": True,
    "code": 10000,
    "message": "操作成功！"
}

login_err_result = {
    "success": False,
    "code": 20001,
    "message": "用户名或密码错误"
}


class TestLogin(unittest.TestCase):

    def test_login_ok(self):
        self.assertTrue(login_ok_result.get('success'))
        self.assertEqual(10000, login_ok_result.get('code'))
        self.assertIn('操作成功', login_ok_result.get('message'))

    def test_login_err(self):
        self.assertFalse(login_err_result.get('success'))
        self.assertEqual(20001, login_err_result.get('code'))
        self.assertIn('用户名或密码错误', login_err_result.get('message'))

